"""
垃圾邮件检测器配置文件
"""

# 模型配置
MODEL_CONFIG = {
    'max_features': 5000,
    'ngram_range': (1, 2),
    'min_df': 1,
    'max_df': 0.95,
    'test_size': 0.2,
    'random_state': 42
}

# 文件路径
PATHS = {
    'model': 'spam_detector_model.pkl',
    'vectorizer': 'vectorizer.pkl',
    'data': 'spam_data.csv'
}

# 邮件类别
LABELS = {
    'ham': 0,
    'spam': 1
}

# 停用词列表（可以自定义）
CUSTOM_STOP_WORDS = [
    'click', 'here', 'now', 'free', 'urgent', 'limited', 'offer'
]